// Copyright (c) 2019 The Bitcoin Core developers
// Distributed under the MIT software license, see the accompanying
// file COPYING or http://www.opensource.org/licenses/mit-license.php.

#ifndef BITCOIN_TEST_UTIL_LOGGING_H
#define BITCOIN_TEST_UTIL_LOGGING_H

#include <util/macros.h>

#include <functional>
#include <list>
#include <string>

class DebugLogHelper {
public:
    //! Custom match checking function.
    //!
    //! Invoked with pointers to lines containing matching strings, and with
    //! null if check_found() is called without any successful match.
    //!
    //! Can return true to enable default DebugLogHelper behavior of:
    //! (1) ending search after first successful match, and
    //! (2) raising an error in check_found if no match was found
    //! Can return false to do the opposite in either case.
    using MatchFn = std::function<bool(const std::string *line)>;

    explicit DebugLogHelper(
        std::string message,
        MatchFn match = [](const std::string *) { return true; });

    DebugLogHelper(const DebugLogHelper &) = delete;
    DebugLogHelper &operator=(const DebugLogHelper &) = delete;

    ~DebugLogHelper();

private:
    const std::string m_message;
    bool m_found{false};
    std::list<std::function<void(const std::string &)>>::iterator
        m_print_connection;
    MatchFn m_match;
};

#define ASSERT_DEBUG_LOG(message)                                              \
    DebugLogHelper UNIQUE_LOG_NAME(debugloghelper)(message)

#endif // BITCOIN_TEST_UTIL_LOGGING_H
